using System.Collections;
using System.Collections.Generic;

namespace IQIGame.Onigao.Framework
{
    /// <summary>
    /// 性能最高的LinkedList
    /// 但不建议业务逻辑使用，因为该类使用上有一些局限，且为了性能，很多高消耗校验都没做。
    /// 
    /// 它有2个问题
    /// 1.实际上它只是组织element之间相互引用，而没有为element创建LinkedListNode来封装，这就意味着，一个element不能同时Add进两个UniqueRefLinkedList。
    /// 2.调用Remove(T node)方法时，请自行注意：node必须已经属于当前UniqueRefLinkedList。
    /// 3.该链表的对象必须实现ILinkedListNode，有一定局限性。
    /// 
    /// 也就意味着性能更高的同时，其应用比较有限。
    /// 
    /// 再次提醒：一个object只能被放进一个UniqueRefLinkedList，同时你也不能用listA.Remove去移除listB中的对象。
    /// </summary>
    /// <typeparam name="T"></typeparam>
    public class UniqueRefLinkedList<T> : IEnumerable<T> where T : class, ILinkedListNode
    {
        private int _count;

        public T head { get; private set; }
        public T tail { get; private set; }
        public int Count => _count;

        // 添加节点到链表头部
        public void AddFirst(T node)
        {
            if (node == null)
            {
                throw new System.Exception("添加节点失败，节点为空。");
            }
            if (node.Previous != null || node.Next != null)
            {
                throw new System.Exception("该节点已经被放入链表，不能被再次放入。");
            }
            if (head == null)
            {
                head = node;
                tail = node;
            }
            else
            {
                head.Previous = node;
                node.Next = head;
                head = node;
            }
            _count++;
        }

        // 添加节点到链表尾部
        public void AddLast(T node)
        {
            if (node == null)
            {
                throw new System.Exception("添加节点失败，节点为空。");
            }
            if (node.Previous != null || node.Next != null)
            {
                throw new System.Exception("该节点已经被放入链表，不能被再次放入。");
            }
            if (head == null)
            {
                head = node;
                tail = node;
            }
            else
            {
                tail.Next = node;
                node.Previous = tail;
                tail = node;
            }
            _count++;
        }

        // 移除特定节点
        public void Remove(T node)
        {
            if (node == null)
            {
                throw new System.Exception("移除节点失败，节点为空。");
            }
            if ((node.Previous == null && node.Previous == null && node != head) || head == null && tail == null)
            {
                return;
            }
            if (node == head)
            {
                head = (T)node.Next;
            }
            else
            {
                node.Previous.Next = node.Next;
            }
            if (node == tail)
            {
                tail = (T)node.Previous;
            }
            else
            {
                node.Next.Previous = node.Previous;
            }
            node.Next = null;
            node.Previous = null;
            _count--;
        }

        // 移除第一个元素并返回
        public T RemoveFirst()
        {
            if (head == null)
            {
                return null;
            }

            T node = head;
            head = (T)head.Next;

            if (head != null)
            {
                head.Previous = null;
            }
            else
            {
                tail = null;
            }

            node.Next = null;
            _count--;
            return node;
        }

        // 移除最后一个元素并返回
        public T RemoveLast()
        {
            if (tail == null)
            {
                return null;
            }

            T node = tail;
            tail = (T)tail.Previous;

            if (tail != null)
            {
                tail.Next = null;
            }
            else
            {
                head = null;
            }

            node.Previous = null;
            _count--;
            return node;
        }

        // 实现 IEnumerable<T>
        public IEnumerator<T> GetEnumerator()
        {
            return new Enumerator(this);
        }

        // 实现 IEnumerable
        IEnumerator IEnumerable.GetEnumerator()
        {
            return new Enumerator(this);
        }

        struct Enumerator : IEnumerator<T>
        {
            private UniqueRefLinkedList<T> list;
            private T current;

            public Enumerator(UniqueRefLinkedList<T> list)
            {
                this.list = list;
                this.current = null;
            }

            public T Current => current;

            object IEnumerator.Current => current;

            public bool MoveNext()
            {
                if (current == null)
                {
                    current = list.head;
                }
                else
                {
                    current = (T)current.Next;
                }
                return current != null;
            }

            public void Reset()
            {
                current = null;
            }

            public void Dispose()
            {
                current = null;
                list = null;
            }
        }
    }
}
